home *** CD-ROM | disk | FTP | other *** search
/ io Programmo 60 / IOPROG_60.ISO / soft / c++ / gsl-1.1.1-setup.exe / {app} / src / poly / zsolve_quadratic.c < prev    next >
Encoding:
C/C++ Source or Header  |  2000-05-05  |  2.2 KB  |  90 lines
  1. /* poly/zsolve_quadratic.c
  2.  * 
  3.  * Copyright (C) 1996, 1997, 1998, 1999, 2000 Brian Gough
  4.  * 
  5.  * This program is free software; you can redistribute it and/or modify
  6.  * it under the terms of the GNU General Public License as published by
  7.  * the Free Software Foundation; either version 2 of the License, or (at
  8.  * your option) any later version.
  9.  * 
  10.  * This program is distributed in the hope that it will be useful, but
  11.  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13.  * General Public License for more details.
  14.  * 
  15.  * You should have received a copy of the GNU General Public License
  16.  * along with this program; if not, write to the Free Software
  17.  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  */
  19.  
  20. /* complex_solve_quadratic.c - finds complex roots of a x^2 + b x + c = 0 */
  21.  
  22. #include <config.h>
  23. #include <math.h>
  24. #include <gsl/gsl_complex.h>
  25. #include <gsl/gsl_poly.h>
  26.  
  27. int
  28. gsl_poly_complex_solve_quadratic (double a, double b, double c,
  29.                   gsl_complex *z0, gsl_complex *z1)
  30. {
  31.   double disc = b * b - 4 * a * c;
  32.  
  33.   if (disc > 0)
  34.     {
  35.       if (b == 0)
  36.     {
  37.       double s = fabs (0.5 * sqrt (disc) / a);
  38.       GSL_REAL (*z0) = -s;
  39.       GSL_IMAG (*z0) = 0;
  40.       GSL_REAL (*z1) = s;
  41.       GSL_IMAG (*z1) = 0;
  42.     }
  43.       else
  44.     {
  45.       double sgnb = (b > 0 ? 1 : -1);
  46.       double temp = -0.5 * (b + sgnb * sqrt (disc));
  47.       double r1 = temp / a;
  48.       double r2 = c / temp;
  49.  
  50.       if (r1 < r2)
  51.         {
  52.           GSL_REAL (*z0) = r1;
  53.           GSL_IMAG (*z0) = 0;
  54.           GSL_REAL (*z1) = r2;
  55.           GSL_IMAG (*z1) = 0;
  56.         }
  57.       else
  58.         {
  59.           GSL_REAL (*z0) = r2;
  60.           GSL_IMAG (*z0) = 0;
  61.           GSL_REAL (*z1) = r1;
  62.           GSL_IMAG (*z1) = 0;
  63.         }
  64.     }
  65.       return 2;
  66.     }
  67.   else if (disc == 0)
  68.     {
  69.       GSL_REAL (*z0) = -0.5 * b / a;
  70.       GSL_IMAG (*z0) = 0;
  71.  
  72.       GSL_REAL (*z1) = -0.5 * b / a;
  73.       GSL_IMAG (*z1) = 0;
  74.  
  75.       return 2;
  76.     }
  77.   else
  78.     {
  79.       double s = fabs (0.5 * sqrt (-disc) / a);
  80.  
  81.       GSL_REAL (*z0) = -0.5 * b / a;
  82.       GSL_IMAG (*z0) = -s;
  83.  
  84.       GSL_REAL (*z1) = -0.5 * b / a;
  85.       GSL_IMAG (*z1) = s;
  86.  
  87.       return 2;
  88.     }
  89. }
  90.